package cc.idoone.web.response;

import org.springframework.core.MethodParameter;
import org.springframework.http.MediaType;
import org.springframework.http.converter.HttpMessageConverter;
import org.springframework.http.server.ServerHttpRequest;
import org.springframework.http.server.ServerHttpResponse;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyAdvice;

import java.util.HashMap;
import java.util.Map;

/**
 * 统计处理返回数据结果
 * <p>
 * Success: ResponseBodyAdvice - HandlerInterceptor<br/>
 * Exception: ExceptionHandler,ResponseBodyAdvice - HandlerExceptionResolver - HandlerInterceptor<br/>
 * </p>
 *
 * @auther hemj
 * @since 2017/12/11
 */
@ControllerAdvice
public class JsonResponseBodyAdvice implements ResponseBodyAdvice<Object> {
	@Override
	public boolean supports(MethodParameter returnType, Class<? extends HttpMessageConverter<?>> converterType) {
		return true;
	}

	@Override
	public Object beforeBodyWrite(Object body,
								  MethodParameter returnType,
								  MediaType selectedContentType,
								  Class<? extends HttpMessageConverter<?>> selectedConverterType,
								  ServerHttpRequest request, ServerHttpResponse response) {
		if (body instanceof Map) {
			Map map = (Map) body;
			if (map.containsKey("result_code")) {
				return map;
			}
		}
		Map resultMap = new HashMap<>();
		resultMap.put("result_code", "0");
		resultMap.put("data", body);
		return resultMap;
	}
}
